from common.litellm import completion_all


def add_to_cart(book_title:str,number:int):
    print(f"{book_title}已入库{number}本书。")

messages = [
    {"role":"system","content":"""
    你是一个书店购物AI助手：你的任务是：
    1、根据用户提供的书名和数量
    2、返回add_to_cart(书名，数量），例如：
add_to_cart("book_title",3)
    3、当信息不完整时，主动询问缺失的书名和数量
    """},
    {"role":"user","content":"我要购买三本书"},
    {"role":"assistant","content":"请问你要购买哪本书？"},
    {"role":"user","content":"chtBi核心技术"}
]

response = completion_all(messages)
resp_text = response.content
if 'add_to_cart' in resp_text:
    eval(resp_text)
print(response)